# -*- coding: utf-8 -*-
# ----------------------------
# @Time    : 2022/6/18 4:12 PM
# @Author  : changqingai
# @FileName: tf_arg.py
# ----------------------------

import tensorflow as tf

# ******* argmax范式 **************
a = tf.random.normal([2, 3], stddev=1, mean=0)
print("a:", a.numpy())
b = tf.argmax(a)
print("b:", b.numpy())

# ******* argmin范式 **************
a = tf.random.normal([2, 3], stddev=1, mean=0)
print("a:", a.numpy())
b = tf.argmin(a)
print("b:", b.numpy())
